package com.sxj.corejava.code02_jdbc;

import com.sxj.corejava.entity.User;

import java.sql.*;

/**
 * @author 石小俊
 * @date 2023年12月2023/12/2日 8:33
 */
public class Test06_保存返回主键 {

    public static void main(String[] args) {
        User user = new User("john", "123456", "13112345678", "江苏-苏州");
        int id = insertReturnPrimaryKey(user);
        System.out.println("保存后返回的主键值为:" + id);
    }

    public static Integer insertReturnPrimaryKey(User user) {
        Connection conn = null;
        PreparedStatement ps = null;
        ResultSet rs = null;
        Integer id = null;
        try {
            Class.forName("com.mysql.jdbc.Driver");
            conn = DriverManager.getConnection("jdbc:mysql://127.0.0.1:3306/jdbc?useUnicode=true&characterEncoding=utf-8", "root", "");
            String sql = new StringBuffer()
                    .append(" insert into ")
                    .append(" t_user ")
                    .append("   (username,password,phone,address)")
                    .append(" values ")
                    .append("   (?,?,?,?)")
                    .toString();
            // 想要实现保存返回主键的功能
            // 在获取状态集的时候,需要设置第二个参数
            // 第二个参数就表示保存后是否需要返回自动生成的键
            // 其值有两个,属于Statement的常量
            // RETURN_GENERATED_KEYS: 返回自动生成的键
            // NO_GENERATED_KEYS: 不返回自动生成的键
            // PreparedStatement是Statement的子类,可以继承对应的常量
            ps = conn.prepareStatement(sql, PreparedStatement.RETURN_GENERATED_KEYS);
            ps.setString(1, user.getUsername());
            ps.setString(2, user.getPassword());
            ps.setString(3, user.getPhone());
            ps.setString(4, user.getAddress());
            // 执行DML操作,即:增删改操作
            ps.executeUpdate();

            // 当设置了保存后返回自动生成的键之后
            // 数据库中执行了保存操作之后,根据id自增的设置会自动生成对应的主键值
            // 此时需要获取到生成的主键的值
            // 其返回值也是一个结果集,但是由于结果集中的数据并不是我们主动查询出来的
            // 因此,其字段名我们可能不知道
            rs = ps.getGeneratedKeys();
            while (rs.next()) {
                // 此时字段名是由数据库厂商为其定义的名字,开发者可能并不知道名字是什么?
                // 1.通过元数据取出对应的名字
                // 2.我们虽然不知道名字是什么,但是我们能确定只有一列
                // 因此,也可以直接通过索引号来获取,即:第几列数据
                id = rs.getInt(1);
            }
        } catch (ClassNotFoundException e) {
            e.printStackTrace();
            System.out.println("未找到驱动");
        } catch (SQLException e) {
            e.printStackTrace();
            System.out.println("连接数据库失败");
        } finally {
            if (conn != null) {
                try {
                    conn.close();
                } catch (SQLException e) {
                    e.printStackTrace();
                    System.out.println("释放连接出错");
                }
            }
            if (ps != null) {
                try {
                    ps.close();
                } catch (SQLException e) {
                    e.printStackTrace();
                }
            }
            if (rs != null) {
                try {
                    rs.close();
                } catch (SQLException e) {
                    e.printStackTrace();
                }
            }
        }
        return id;
    }

}
